Armstrong number Python program